/*
 * Copyright (c) 2010, VeRSI Consortium
 *   (Victorian eResearch Strategic Initiative, Australia)
 * All rights reserved.
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in the
 *       documentation and/or other materials provided with the distribution.
 *     * Neither the name of the VeRSI, the VeRSI Consortium members, nor the
 *       names of its contributors may be used to endorse or promote products
 *       derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND ANY
 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE REGENTS AND CONTRIBUTORS BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
package versi.vitro.oaiserviceprovider.service.jena.impl;

import java.util.HashMap;
import java.util.Map;

import org.apache.log4j.Logger;

import versi.vitro.oaiserviceprovider.service.jena.api.ModelBuilder;

import com.hp.hpl.jena.ontology.OntModel;
import com.hp.hpl.jena.ontology.OntModelSpec;
import com.hp.hpl.jena.rdf.model.Model;
import com.hp.hpl.jena.rdf.model.ModelFactory;
import com.hp.hpl.jena.rdf.model.ModelMaker;

import edu.cornell.mannlib.vitro.webapp.dao.jena.JenaBaseDaoCon;

/**
 * Service implementing {@link ModelBuilder}.
 */
public class ModelBuilderImpl implements ModelBuilder {

    /** log4j instance. */
    private static final Logger LOGGER = Logger.getLogger(ModelBuilderImpl.class);
    
    private interface OntologyModelFactory {
	Model create();
    }

    public static OntModelSpec MEM_ONT_MODEL_SPEC = OntModelSpec.OWL_MEM;

    /** RDF Model Maker. */
    private ModelMaker modelMaker;

    private Map<String, OntologyModelFactory> modelFactoryLookUp;

    /**
     * Default Constructor.
     * 
     * @param modelMaker modelMaker.
     */
    public ModelBuilderImpl(ModelMaker modelMaker) {
	super();
	this.modelMaker = modelMaker;
	modelFactoryLookUp = new HashMap<String, OntologyModelFactory>();
	modelFactoryLookUp.put("vitro:jenaOntModel", new JenaOntologyModelFactory());
	modelFactoryLookUp.put("vitro:baseOntModel", new BaseOntologyModelFactory());
	modelFactoryLookUp.put("vitro:inferenceOntModel", new InferenceOntologyModelFactory());
    }

    /** {@inheritDoc} */
    public Model createOntologyModelFromName(final String modelName) {

	// String defaultNamespace = DEFAULT_DEFAULT_NAMESPACE;
	OntologyModelFactory ontologyModelFactory = modelFactoryLookUp.get(modelName);
	if (ontologyModelFactory != null) {
	    return ontologyModelFactory.create();
	}
	return modelMaker.getModel(modelName);
    }

    /** Jena ontology Model class. */
    private static class JenaOntologyModelFactory implements OntologyModelFactory {
	public Model create() {
	    Model memModel = new BaseOntologyModelFactory().create();
	    Model inferenceModel = new InferenceOntologyModelFactory().create();
	    Model unionModel = ModelFactory.createUnion(memModel, inferenceModel);
	    OntModel memoryModel = ModelFactory.createOntologyModel(MEM_ONT_MODEL_SPEC, unionModel);
	    return memoryModel;
	}
    }

    /** Inference ontology Model class. */
    private static class InferenceOntologyModelFactory implements OntologyModelFactory {
	public Model create() {
	    OntModel inferenceModel = ModelFactory.createOntologyModel(MEM_ONT_MODEL_SPEC);
	    return inferenceModel;
	}
    }

    /** Base ontology Model class. */
    private static class BaseOntologyModelFactory implements OntologyModelFactory {
	public Model create() {
	    OntModel memModel = ModelFactory.createOntologyModel(MEM_ONT_MODEL_SPEC);
	    memModel.addSubModel((new JenaBaseDaoCon()).getConstModel());
	    return memModel;
	}
    }

    /** {@inheritDoc} */
    public OntModel createOntologyModelFromSpec(final OntModelSpec ontologyModelSpecification) {
	return ModelFactory.createOntologyModel(ontologyModelSpecification);
    }
}
